package com.zm.demo.core.leetcode;

import javafx.util.Pair;

import java.util.Stack;

/**
 * 二叉树的最小深度
 * 给定一个二叉树，找出其最小深度。

 最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

 说明: 叶子节点是指没有子节点的节点。

 示例:

 给定二叉树 [3,9,20,null,null,15,7],

  3
 / \
 9  20
    /  \
   15   7
 返回它的最小深度  2.

 来源：力扣（LeetCode）
 链接：https://leetcode-cn.com/problems/minimum-depth-of-binary-tree
 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * Created by Administrator on 2020/1/11.
 */
public class MinDepth {

    public int minDepth(TreeNode root) {
        if (root == null){
            return 0;
        }else {
            return minDepthDFS(root, 1);
        }
    }

    public int minDepthDFS(TreeNode root, int level) {
        if (root.left == null && root.right == null){
            return level;
        }
        if (root.left == null){
            return minDepthDFS(root.right, level+1);
        }
        if (root.right == null){
            return minDepthDFS(root.left, level+1);
        }
        return Math.min(minDepthDFS(root.left, level+1),minDepthDFS(root.right, level+1));

    }

    public int minDepthBFS(TreeNode root) {

        if (root == null){
            return 0;
        }
        Stack<Pair<TreeNode, Integer>> stack = new Stack<>();
        stack.push(new Pair<>(root, 1));
        while (!stack.isEmpty()){
            Pair<TreeNode,Integer> pair = stack.pop();
            TreeNode treeNode = pair.getKey();
            if (treeNode.left == null && treeNode.right == null){
                break;
            }
            if (treeNode.left != null){
                stack.push(new Pair<>(treeNode.left, pair.getValue()+1));
            }

            if (treeNode.right != null){
                stack.push(new Pair<>(treeNode.right, pair.getValue()+1));
            }
        }
        return stack.pop().getValue();

    }

}
